package com.wwl.cms.utils.designModeTest.flyweight;

import java.util.HashMap;

/**
 * @author wwl
 * @version 1.0
 * @date 2021/9/8 18:02
 * @description:
 */
//网站工厂类,根据需要返回下一个网站
public class WebSiteFactory {
    //集合，充当池的作用
    private HashMap<String,ConcreteWebSite> pool = new HashMap<>();
    //根据网站的类型，返回一个网站，如果没有就创建一个网站，并放入到池中并返回
    public WebSite getWebSiteCategory(String type){
        if (!pool.containsKey(type)){
            //就创建一个网站，并放入到池中
            pool.put(type,new ConcreteWebSite(type));
        }
        return pool.get(type);
    }
    //获取网站分类的总数
    public int getWebSiteCount(){
        return pool.size();
    }
}
